package com.bestv.search.common.util;

import org.apache.commons.configuration.PropertiesConfiguration;
import org.apache.commons.configuration.reloading.FileChangedReloadingStrategy;

import java.io.UnsupportedEncodingException;

/**
 * @author Michael Wang
 */
public class ChineseCharToEn {

    // 国标码和区位码转换常量
    private static final int GB_SP_DIFF = 160;

    /**
     * 每个声母在汉字区位码表中的起始位置
     */
    private final static int[] li_SecPosValue = {1601, 1637, 1833, 2078, 2274, 2302, 2433, 2594, 2787, 3106, 3212,
            3472, 3635, 3722, 3730, 3858, 4027, 4086, 4390, 4558, 4684, 4925, 5249, 5590};
    private final static String[] lc_FirstLetter = {"a", "b", "c", "d", "e", "f", "g", "h", "j", "k", "l", "m", "n",
            "o", "p", "q", "r", "s", "t", "w", "x", "y", "z"};

    private final static String[] punctuation = {"·", "—"};
    private final static String chineseFile = "chinese.properties";

    /**
     * 取得给定汉字串的首字母串,即声母串
     *
     * @param str 给定汉字串
     * @return 声母串
     */
    public static String getAllFirstLetter(String str) {
        if (str == null || str.trim().length() == 0) {
            return "";
        }

        String _str = "";
        for (int i = 0; i < str.length(); i++) {
            _str = _str + getFirstLetter(str.substring(i, i + 1));
        }

        return _str;
    }

    /**
     * 取得给定汉字的首字母,即声母 GB码两个字节分别减去160，转换成10进制码组合就可以得到区位码
     *
     * @param str 给定的汉字
     * @return 给定汉字的声母
     */
    public static String getFirstLetter(String str) {
        if (str == null) {
            return "";
        }
        if (str.trim().length() == 0) {
            return str;
        }
        String chinese = conversionStr(str, "GBK", "ISO8859-1");

        if (chinese.length() > 1) // 判断是不是汉字
        {
            int li_SectorCode = (int) chinese.charAt(0); // 汉字区码
            int li_PositionCode = (int) chinese.charAt(1); // 汉字位码
            li_SectorCode = li_SectorCode - GB_SP_DIFF;
            li_PositionCode = li_PositionCode - GB_SP_DIFF;
            int li_SecPosCode = li_SectorCode * 100 + li_PositionCode; // 汉字区位码
            if (li_SecPosCode > 1600 && li_SecPosCode < 5590) {
                for (int i = 0; i < 23; i++) {
                    if (li_SecPosCode >= li_SecPosValue[i] && li_SecPosCode < li_SecPosValue[i + 1]) {
                        chinese = lc_FirstLetter[i];
                        break;
                    }
                }
            } else // 非GB表一级汉字
            {
                chinese = str;
                chinese = secondLevelChinese(chinese);
            }
        } else {
            if (isPunc(str))
                return " ";
        }
        return chinese;
    }

    private static String secondLevelChinese(String chinese) {
        PropertiesConfiguration config;
        try {
            config = new PropertiesConfiguration(chineseFile);
            config.setReloadingStrategy(new FileChangedReloadingStrategy());
            for (int i = 0; i < lc_FirstLetter.length; i++) {
                String fl = lc_FirstLetter[i];
                if (config.containsKey(fl)) {
                    String value = config.getString(fl);
                    if (value.indexOf(chinese) != -1)
                        return fl;
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            return chinese;
        }
        return chinese;
    }

    /**
     * 字符串编码转换
     *
     * @param str           要转换编码的字符串
     * @param charsetName   原来的编码
     * @param toCharsetName 转换后的编码
     * @return 经过编码转换后的字符串
     */
    private static String conversionStr(String str, String charsetName, String toCharsetName) {
        try {
            str = new String(str.getBytes(charsetName), toCharsetName);
            // System.out.println(str);
        } catch (UnsupportedEncodingException ex) {
            System.out.println("字符串编码转换异常：" + ex.getMessage());
        }

        return str;
    }

    private static boolean isPunc(String punc) {
        for (int i = 0; i < punctuation.length; i++) {
            String p = punctuation[i];
            if (punc.equals(p)) {
                return true;
            }
        }
        return false;
    }
}